/*!
 * \file __frproto_pri_add.c
 *
 * All Rights Reserved Copyright (C) 2014 broada,Co.,Ltd.\n
 *
 * \version 3.0
 * \author geyh
 * \date 2016/03/31
 *
 */
#include <CUnit/Basic.h>
#include <stdlib.h>

#include "frcomm.h"
#include "frmem.h"
#include "frlist.h"
#include "frproto_mng.h"
#include "apmtest_cunit.h"

#include "apmtest_comm.h"

static int __suite_init(void)
{
//    frmodule_mng_init();
    return 0;
}

static int __suite_cleanup(void)
{
//    frmodule_mng_destory();
    return 0;
}


typedef struct __st_proto_priority {
    u_short us_port_start;      /* port 开始 */
    u_short us_port_end;        /* port 结束 */

    Frmodule_mng *p_module_mng;
} Frproto_priority;

void __list_node_priority_free(Frlist_node *p_st_node);
int __frproto_pri_add(Frlist_header *p_st_hdr, u_short us_port_start, u_short us_port_end, int i_l7_proto);

static void __case1(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frlist_header *p_st_hdr = NULL;
    u_short us_port_start = 0;
    u_short us_port_end = 0;
    int i_l7_proto = 1;
    int i_ret = 0;

    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);
    p_st_hdr = frlist_create();

    frproto_mng_init();
    i_ret = __frproto_pri_add(p_st_hdr, us_port_start, us_port_end, i_l7_proto);

    us_port_start = 20;
    us_port_end = 80;
    i_ret = __frproto_pri_add(p_st_hdr, us_port_start, us_port_end, i_l7_proto);
    CU_ASSERT_EQUAL(i_ret,FR_SUCC);

    frmodule_mng_destory();
    frlist_destroy(p_st_hdr, __list_node_priority_free);
    frproto_mng_destory();

}

static void __case2(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frlist_header *p_st_hdr = NULL;
    u_short us_port_start = 0;
    u_short us_port_end = 0;
    int i_l7_proto = 2;
    int i_ret = 0;

    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);
    p_st_hdr = frlist_create();

    frproto_mng_init();
    i_ret = __frproto_pri_add(p_st_hdr, us_port_start, us_port_end, i_l7_proto);

    us_port_start = 20;
    us_port_end = 80;
    i_l7_proto = 1;
    i_ret = __frproto_pri_add(p_st_hdr, us_port_start, us_port_end, i_l7_proto);
    CU_ASSERT_EQUAL(i_ret,FR_SUCC);

    frmodule_mng_destory();
    frlist_destroy(p_st_hdr, __list_node_priority_free);
    frproto_mng_destory();

}


int add_suite___frproto_pri_add(void)
{
    CU_pSuite pSuite = NULL;

    if ((pSuite = CU_add_suite("frproto_mng __frproto_pri_add()", __suite_init, __suite_cleanup)) == NULL)
        return FR_FAIL;

    CU_add_test(pSuite, "case1", __case1);
    CU_add_test(pSuite, "case2", __case2);
//    CU_add_test(pSuite, "case3", __case3);
//    CU_add_test(pSuite, "case4", __case4);
//    CU_add_test(pSuite, "case5", __case5);

    return FR_SUCC;
}
